pass lambda function in python